import os

import dotenv

dotenv.load_dotenv()

_MODEL_PATH_DICT = {
    'meta-llama/Llama-3.1-8B': os.environ['MODEL_LLAMA_3_1_8B']
}

def transform_model_path(model_name: str) -> str:
    if model_name in _MODEL_PATH_DICT:
        return _MODEL_PATH_DICT[model_name]
    return model_name